from gym.envs.registration import register


register(
    id='puck-world-v0',
    entry_point='puck_world.envs.single_agent.puckworld_continuous:PuckWorld'
)

register(
    id='puck-world-wheel-v0',
    entry_point='puck_world.envs.multi_agents.puckworld_continuous:PuckWorldWheel'
)